import numpy as np
from sklearn.linear_model import LinearRegression
import matplotlib.pyplot as plt
# 五个楼盘，分别距离市中心5公里，10公里, 12公里, 13公里, 15公里
x = np.array([[5], [10], [12], [13], [15]])
# 价格分别为20w，15w，8w，6w，4w
y = np.array([20, 15, 8, 6, 4])

# plt.scatter(x, y, color='black')
# plt.show()

# 模型训练，自动调用梯度下降,往往只能处理简单模型
reg = LinearRegression().fit(x, y)
# 斜率
print(reg.coef_)
# 截距
print(reg.intercept_)
# 进行评分 满分1分
print(reg.score(x, y))
# 预测 给一个新的x值，预测y是多少。注：距离市中心3公里的房子多少钱
print(reg.predict(np.array([[3]])))